package leetcode.editor.cn;

//给定一个二叉树，判断它是否是高度平衡的二叉树。 
//
// 本题中，一棵高度平衡二叉树定义为： 
//
// 
// 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。 
// 
//
// 
//
// 示例 1： 
//
// 
//输入：root = [3,9,20,null,null,15,7]
//输出：true
// 
//
// 示例 2： 
//
// 
//输入：root = [1,2,2,3,3,null,null,4,4]
//输出：false
// 
//
// 示例 3： 
//
// 
//输入：root = []
//输出：true
// 
//
// 
//
// 提示： 
//
// 
// 树中的节点数在范围 [0, 5000] 内 
// -10⁴ <= Node.val <= 10⁴ 
// 
// Related Topics 树 深度优先搜索 二叉树 👍 935 👎 0


import com.liushi.util.TreeNode;

// Java：平衡二叉树
public class P110BalancedBinaryTree {
    public static void main(String[] args) {
        Solution solution = new P110BalancedBinaryTree().new Solution();
        // TO TEST
    }
    //leetcode submit region begin(Prohibit modification and deletion)

    /**
     * Definition for a binary tree node.
     * public class TreeNode {
     * int val;
     * TreeNode left;
     * TreeNode right;
     * TreeNode() {}
     * TreeNode(int val) { this.val = val; }
     * TreeNode(int val, TreeNode left, TreeNode right) {
     * this.val = val;
     * this.left = left;
     * this.right = right;
     * }
     * }
     */
    class Solution {
        public boolean isBalanced(TreeNode root) {
            return process(root).isBalance;
        }

        /**
         *
         */
        public Info process(TreeNode head) {
            if (head == null) {
                return new Info(true, 0);
            }

            Info leftInfo = process(head.left);
            Info rightInfo = process(head.right);

            // 左树上的高度和右树上的高度最大值, 在加上我自己这一层, 就是我的高度
            int height = Math.max(leftInfo.height, rightInfo.height) + 1;
            boolean isBalanced = false;
            if (leftInfo.isBalance && rightInfo.isBalance && Math.abs(leftInfo.height - rightInfo.height) < 2) {
                isBalanced = true;
            }

            return new Info(isBalanced, height);
        }

        // 左右子树要求一样, 信息返回的结构体
        public class Info {
            public boolean isBalance;
            public int height;

            public Info(boolean isBalance, int height) {
                this.isBalance = isBalance;
                this.height = height;
            }
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}
